import unittest
import numpy as np

from Utilities.MathFunctions import polynom

class MyTestCase(unittest.TestCase):

    def test_func(self):
        re_database = [2, 2.5, 3, 4]
        lambda_database = [0.032, 0.034, 0.04, 0.04]
        re_database = np.array(re_database) * 10 ** 3

        self.assertEqual(round(polynom(2500, re_database, lambda_database, 4), 4),
                         0.034)


if __name__ == '__main__':
    unittest.main()
